import os

# 定义函数来修改文件的最后五行
def modify_last_lines(file_path, new_lines):
    with open(file_path, 'r+') as file:
        lines = file.readlines()
        # 确保文件至少有五行
        if len(lines) >= 5:
            # 保留除最后五行之外的所有行
            lines = lines[:-5]
            # 添加新的五行到文件末尾
            lines.extend(new_lines)
            # 移动到文件开头并清空内容
            file.seek(0)
            file.truncate()
            # 写入修改后的所有行
            file.writelines(lines)

# 要替换的新五行内容
new_content = [
    "Y0.000\n",
    "X0.000\n",
    "G00Z10.000\n",
    "M5\n",
    "M02\n"
]

# 遍历当前目录下的所有文本文件
for filename in os.listdir('.'):
    if filename.endswith('.txt'):  # 假设文件扩展名为.txt
        modify_last_lines(filename, new_content)
        print(f"Modified {filename}")

print("All files have been modified.")